D2GRs2 源码阅读1-概述

GitHub-generative-recommenders).

├── configs                  # 实验配置 数据、结构、参数等
│   ├── amzn-books           # Amazon Reviews(base-SASRec)
│   ├── ml-1m                # MovieLens 1m
│   └── ml-20m               # MovieLens 20m
├── data
│   ├── dataset.py           # 构建序列 dataset 时间逆序
│   ├── eval.py              # 评估指标，并记录到TensorBoard
│   ├── item_features.py     # 定义`ItemFeatures`数据类
│   ├── preprocessor.py      # 预处理两种数据集(下载、处理等)
│   └── reco_dataset.py      # 获取推荐（训练、评估）数据集
├── indexing
│   ├── __init__.py
│   ├── candidate_index.py   # 构建和处理推荐候选索引和Top-K
│   ├── mips_top_k.py        # 计算输入item和候选的topk
│   └── utils.py             # 工厂函数，切换不同Top-K检索算法
├── modeling
│   ├── __init__.py
│   ├── sequential           # *模型核心代码*
│   ├── similarity           # *MoL网络模块，高效的相似度计算*
│   ├── initialization.py    # 网络权重初始化
│   ├── ndp_module.py        # 网络模块抽象
│   ├── similarity_module.py # 用户物品交互的embedding的组件？
│   └── similarity_utils.py  # 创建和返回不同的相似度计算模块
├── ops
│   └── triton               # Triton性能优化
├── tests                    # 单元测试脚本
│   └── ragged_hstu_attention_test.py
├── trainer
│    └── data_loader.py      # 数据加载器、支持分布式训练
│ 
├── preprocess_public_data.py# 数据预处理入口脚本
├── train.py                 # 训练入口脚本
│ 
├── CODE_OF_CONDUCT.md       # 参与者行为准则
├── CONTRIBUTING.md          # 贡献
├── LICENSE                  # 开源 Apache 2.0 允许商用
├── requirements.txt
├── README.md

brew install tree
tree -L 2